#！/usr/bin/python
# _*_ coding:UTF-8 -*-
import pandas as pd

def MA(df, n,ksgn='close'):  
    '''
    def MA(df, n,ksgn='close'):  
    #Moving Average  
    MA是简单平均线，也就是平常说的均线
    【输入】
        df, pd.dataframe格式数据源
        n，时间长度
        ksgn，列名，一般是：close收盘价
    【输出】    
        df, pd.dataframe格式数据源,
        增加了一栏：ma_{n}，均线数据
    '''
    xnam='ma_{n}'.format(n=n)
    #ds5 = pd.Series(pd.rolling_mean(df[ksgn], n), name =xnam)  
    ds2=pd.Series(df[ksgn], name =xnam);
    ds5 = ds2.rolling(center=False,window=n).mean() 
    #print(ds5.head()); print(df.head())
    df = df.join(ds5)  
    
    return df

def rate(df, ksgn='close'):
    '''
    rate 函数计算当天的涨幅，以百分百形式展示
    【输入，输出】
        df: pd.dataframe格式数据源
        输出增加一列
    '''
    ds2= pd.Series(df[ksgn],name='rate')
    ds5 = ds2.pct_change()*100
    df = df.join(ds5)
    return df